package com.example.utils.util;

import org.springframework.web.multipart.MultipartFile;

import java.io.File;

/**
 * TODO  上传文件
 *
 * @ClassName Upload
 * @Author xiaojiang
 * @Date 2021/9/26 下午 15:13
 */
public class Upload {

    /**
     *  TODO  上传文件
     * @Param : filePath
     * @Param : file
     * @return String  0 = 成功  ; 1 = 失败
     * @Author 江
     * @Date 2021/9/26 下午 15:16
     */
    public static String upload(String filePath, MultipartFile file) {
        // 获取文件名
        String fileName = file.getOriginalFilename();
        System.out.println("上传的文件名为：" + fileName);
        if (!typeUtil(fileName)){
            return "1";
        }
        File dest = new File(filePath + fileName);
        // 检测是否存在目录
        if (!dest.exists()) {
            dest.mkdirs();
        }
        try {
            file.transferTo(dest);
            return dest.toString();
        } catch (Exception e) {
            e.printStackTrace();
            return "0";
        }
    }

    /**
     *  TODO  判断是否是需要的文件名，也可以判断文件类型或者图片类型
     * @Param : fileName  文件名
     * @return boolean
     * @Author 江
     * @Date 2021/9/26 下午 15:15
     */
    public static boolean typeUtil(String fileName){
        String name[] = { "5GC_Sim_AMF_Table.db", "5GC_Sim_gNB_Table.db", "5GC_Sim_NSSAI_Table.db",
                "5GC_Sim_PDN_Table.db", "5GC_Sim_QoS_Info_Table.db", "5GC_Sim_UE_Auth_Info_Table.db",
                "5GC_Sim_UE_Table.db", "5GC_Sim_UPF_Table.db"};
        for (int i = 0; i < name.length; i++) {
            if (name[i].equals(fileName)) {
                return true;
            }
        }
        return false;

    }



}
